/*
 * libusb stress test program to perform simple stress tests
 * Copyright © 2012 Toby Gray <toby.gray@realvnc.com>
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */

#include <stdio.h>
#include <string.h>
#include <memory.h>

#include "libusb.h"
#include "libusb_testlib.h"

/** Test that creates and destroys a single concurrent context
 * 10000 times. */
static libusb_testlib_result test_init_and_exit(libusb_testlib_ctx * tctx)
{
    libusb_context * ctx = NULL;
    int i;
    for (i = 0; i < 10000; ++i) {
        int r = libusb_init(&ctx);
        if (r != LIBUSB_SUCCESS) {
            libusb_testlib_logf(tctx,
                "Failed to init libusb on iteration %d: %d",
                i, r);
            return TEST_STATUS_FAILURE;
        }
        libusb_exit(ctx);
        ctx = NULL;
    }

    return TEST_STATUS_SUCCESS;
}

/** Tests that devices can be listed 1000 times. */
static libusb_testlib_result test_get_device_list(libusb_testlib_ctx * tctx)
{
    libusb_context * ctx = NULL;
    int r, i;
    r = libusb_init(&ctx);
    if (r != LIBUSB_SUCCESS) {
        libusb_testlib_logf(tctx, "Failed to init libusb: %d", r);
        return TEST_STATUS_FAILURE;
    }
    for (i = 0; i < 1000; ++i) {
        libusb_device ** device_list;
        ssize_t list_size = libusb_get_device_list(ctx, &device_list);
        if (list_size < 0 || device_list == NULL) {
            libusb_testlib_logf(tctx,
                "Failed to get device list on iteration %d: %d (%p)",
                i, -list_size, device_list);
            return TEST_STATUS_FAILURE;
        }
        libusb_free_device_list(device_list, 1);
    }
    libusb_exit(ctx);
    return TEST_STATUS_SUCCESS;
}

/** Tests that 100 concurrent device lists can be open at a time. */
static libusb_testlib_result test_many_device_lists(libusb_testlib_ctx * tctx)
{
#define LIST_COUNT 100
    libusb_context * ctx = NULL;
    libusb_device ** device_lists[LIST_COUNT];
    int r, i;
    memset(device_lists, 0, sizeof(device_lists));

    r = libusb_init(&ctx);
    if (r != LIBUSB_SUCCESS) {
        libusb_testlib_logf(tctx, "Failed to init libusb: %d", r);
        return TEST_STATUS_FAILURE;
    }

    /* Create the 100 device lists. */
    for (i = 0; i < LIST_COUNT; ++i) {
        ssize_t list_size = libusb_get_device_list(ctx, &(device_lists[i]));
        if (list_size < 0 || device_lists[i] == NULL) {
            libusb_testlib_logf(tctx,
                "Failed to get device list on iteration %d: %d (%p)",
                i, -list_size, device_lists[i]);
            return TEST_STATUS_FAILURE;
        }
    }

    /* Destroy the 100 device lists. */
    for (i = 0; i < LIST_COUNT; ++i) {
        if (device_lists[i]) {
            libusb_free_device_list(device_lists[i], 1);
            device_lists[i] = NULL;
        }
    }

    libusb_exit(ctx);
    return TEST_STATUS_SUCCESS;
#undef LIST_COUNT
}

/** Tests that the default context (used for various things including
 * logging) works correctly when the first context created in a
 * process is destroyed. */
static libusb_testlib_result test_default_context_change(libusb_testlib_ctx * tctx)
{
    libusb_context * ctx = NULL;
    int r, i;

    for (i = 0; i < 100; ++i) {
        /* First create a new context */
        r = libusb_init(&ctx);
        if (r != LIBUSB_SUCCESS) {
            libusb_testlib_logf(tctx, "Failed to init libusb: %d", r);
            return TEST_STATUS_FAILURE;
        }

        /* Enable debug output, to be sure to use the context */
        libusb_set_debug(NULL, LIBUSB_LOG_LEVEL_DEBUG);
        libusb_set_debug(ctx, LIBUSB_LOG_LEVEL_DEBUG);

        /* Now create a reference to the default context */
        r = libusb_init(NULL);
        if (r != LIBUSB_SUCCESS) {
            libusb_testlib_logf(tctx, "Failed to init libusb: %d", r);
            return TEST_STATUS_FAILURE;
        }

        /* Destroy the first context */
        libusb_exit(ctx);
        /* Destroy the default context */
        libusb_exit(NULL);
    }

    return TEST_STATUS_SUCCESS;
}

/* Fill in the list of tests. */
static const libusb_testlib_test tests[] = {
    {"init_and_exit", &test_init_and_exit},
    {"get_device_list", &test_get_device_list},
    {"many_device_lists", &test_many_device_lists},
    {"default_context_change", &test_default_context_change},
    LIBUSB_NULL_TEST
};

int main (int argc, char ** argv)
{
    return libusb_testlib_run_tests(argc, argv, tests);
}

